Mercurial

comparison connectors/tests/core_store_test.c @ 279:b3b547563ec7

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add Google connector service and agent wiki Implement the C/Seobeo Google Drive and Gmail connector with encrypted OAuth storage, Zenbu authentication, browser testing, AI tool discovery, chunked HTTP decoding, and Bazel coverage. Consolidate repository guidance into progressive wiki documentation and enforce arena-first allocation for new first-party C code. Co-authored-by: Copilot <[email protected]> Copilot-Session: 84c338fd-0939-4bb3-b7f3-1062eb213e5d
author MrJuneJune <me@mrjunejune.com>
date Mon, 17 Aug 2026 22:22:36 -0700
parents
children
comparison
equal deleted inserted replaced
278:8d560f50ed4c 279:b3b547563ec7
1 #include "connectors/connector.h"
2
3 #include <assert.h>
4 #include <stdio.h>
5 #include <string.h>
6
7 static Connector_Master_Key key_with(uint8 value, uint32 version)
8 {
9 Connector_Master_Key key = {.version = version};
10 memset(key.key, value, sizeof(key.key));
11 return key;
12 }
13
14 static Connector_Account account_for(
15 const char *user, const char *account_id, const char *subject)
16 {
17 Connector_Account account;
18 memset(&account, 0, sizeof(account));
19 strcpy(account.account_id, account_id);
20 strcpy(account.user_id, user);
21 strcpy(account.provider, "google");
22 strcpy(account.provider_subject, subject);
23 strcpy(account.email, "[email protected]");
24 strcpy(account.access_token, "access-secret");
25 strcpy(account.refresh_token, "refresh-secret");
26 strcpy(account.scopes, "scope");
27 account.expires_at = 123456;
28 return account;
29 }
30
31 static void test_crypto(void)
32 {
33 Connector_Master_Key key = key_with(7, 3);
34 char encrypted[4096], decrypted[128];
35 assert(Connector_Encrypt(&key, "highly-secret", encrypted, sizeof(encrypted)));
36 assert(strcmp(encrypted, "highly-secret") != 0);
37 assert(Connector_Decrypt(&key, encrypted, decrypted, sizeof(decrypted)));
38 assert(!strcmp(decrypted, "highly-secret"));
39
40 Connector_Master_Key wrong = key_with(8, 3);
41 assert(!Connector_Decrypt(&wrong, encrypted, decrypted, sizeof(decrypted)));
42 Connector_Master_Key wrong_version = key;
43 wrong_version.version = 4;
44 assert(!Connector_Decrypt(
45 &wrong_version, encrypted, decrypted, sizeof(decrypted)));
46 encrypted[strlen(encrypted) - 1] =
47 encrypted[strlen(encrypted) - 1] == 'A' ? 'B' : 'A';
48 assert(!Connector_Decrypt(&key, encrypted, decrypted, sizeof(decrypted)));
49 }
50
51 static void test_store(void)
52 {
53 Connector_Master_Key key = key_with(17, 1);
54 Connector_Store store;
55 assert(Connector_Store_Open(&store, ":memory:", &key));
56
57 Connector_OAuth_Start state = {0};
58 strcpy(state.state, "state-1");
59 strcpy(state.code_verifier, "verifier-1");
60 assert(Connector_Store_Create_State(&store, "user-a", &state, 100));
61 char verifier[128];
62 assert(!Connector_Store_Consume_State(
63 &store, "user-a", "state-1", 101, verifier, sizeof(verifier)));
64 strcpy(state.state, "state-2");
65 assert(Connector_Store_Create_State(&store, "user-a", &state, 200));
66 assert(!Connector_Store_Consume_State(
67 &store, "user-b", "state-2", 150, verifier, sizeof(verifier)));
68 assert(Connector_Store_Consume_State(
69 &store, "user-a", "state-2", 150, verifier, sizeof(verifier)));
70 assert(!strcmp(verifier, "verifier-1"));
71 assert(!Connector_Store_Consume_State(
72 &store, "user-a", "state-2", 150, verifier, sizeof(verifier)));
73
74 Connector_Account account = account_for("user-a", "google:one", "one");
75 assert(Connector_Store_Save_Account(&store, &account));
76 Connector_Account second = account_for("user-a", "google:two", "two");
77 strcpy(second.email, "[email protected]");
78 assert(Connector_Store_Save_Account(&store, &second));
79 Connector_Account loaded;
80 assert(Connector_Store_Get_Account(
81 &store, "user-a", "google:one", &loaded));
82 assert(!strcmp(loaded.access_token, "access-secret"));
83 assert(!Connector_Store_Get_Account(
84 &store, "user-b", "google:one", &loaded));
85 assert(!Connector_Store_Delete_Account(
86 &store, "user-b", "google:one"));
87 assert(Connector_Store_Get_Account(
88 &store, "user-a", "google:two", &loaded));
89 assert(!strcmp(loaded.email, "[email protected]"));
90 Dowa_Arena *list_arena = Dowa_Arena_Create(64 * 1024);
91 Connector_Account_Summary *accounts = NULL;
92 assert(Connector_Store_List_Accounts(
93 &store, "user-a", &accounts, list_arena));
94 assert(Dowa_Array_Length(accounts) == 2);
95 assert(!strcmp(accounts[0].account_id, "google:one"));
96 assert(!strcmp(accounts[1].account_id, "google:two"));
97 Dowa_Arena_Free(list_arena);
98 Dowa_Arena *query_arena = Dowa_Arena_Create(4096);
99 Deita_Result_Set *ciphertext = Deita_Query_Execute(
100 store.connection,
101 "SELECT access_token_enc FROM connector_accounts "
102 "WHERE account_id='google:one'",
103 query_arena);
104 assert(ciphertext && Deita_Result_Set_Next(ciphertext));
105 assert(!strstr(Deita_Result_Set_Get_Text(ciphertext, 0), "access-secret"));
106 Deita_Result_Set_Free(ciphertext);
107 Dowa_Arena_Free(query_arena);
108
109 assert(Connector_Store_Get_Policy(
110 &store, "user-a", "gmail.send") == CONNECTOR_CONFIRM_ALWAYS);
111 assert(Connector_Store_Get_Policy(
112 &store, "user-a", "drive.overwrite") == CONNECTOR_CONFIRM_ALWAYS);
113 assert(Connector_Store_Get_Policy(
114 &store, "user-a", "drive.create") == CONNECTOR_CONFIRM_NEVER);
115 assert(Connector_Store_Set_Policy(
116 &store, "user-a", "gmail.send", CONNECTOR_CONFIRM_NEVER));
117 assert(Connector_Store_Get_Policy(
118 &store, "user-a", "gmail.send") == CONNECTOR_CONFIRM_NEVER);
119
120 assert(Connector_Store_Create_Confirmation(
121 &store, "user-a", "digest-a", "confirm-a", 300));
122 assert(!Connector_Store_Consume_Confirmation(
123 &store, "user-b", "digest-a", "confirm-a", 250));
124 assert(!Connector_Store_Consume_Confirmation(
125 &store, "user-a", "digest-b", "confirm-a", 250));
126 assert(Connector_Store_Consume_Confirmation(
127 &store, "user-a", "digest-a", "confirm-a", 250));
128 assert(!Connector_Store_Consume_Confirmation(
129 &store, "user-a", "digest-a", "confirm-a", 250));
130 assert(Connector_Store_Create_Confirmation(
131 &store, "user-a", "digest-c", "confirm-c", 300));
132 assert(!Connector_Store_Consume_Confirmation(
133 &store, "user-a", "digest-c", "confirm-c", 301));
134
135 assert(Connector_Store_Put_Idempotent(
136 &store, "user-a", "idem-1", "digest-1", 201, "{\"id\":\"x\"}"));
137 int32 status = 0;
138 char body[128];
139 assert(Connector_Store_Get_Idempotent(
140 &store, "user-a", "idem-1", "digest-1", &status, body, sizeof(body)));
141 assert(status == 201 && !strcmp(body, "{\"id\":\"x\"}"));
142 assert(!Connector_Store_Get_Idempotent(
143 &store, "user-a", "idem-1", "digest-other",
144 &status, body, sizeof(body)));
145 assert(!Connector_Store_Get_Idempotent(
146 &store, "user-b", "idem-1", "digest-1", &status, body, sizeof(body)));
147 assert(Connector_Store_Idempotency_Conflict(
148 &store, "user-a", "idem-1", "digest-other"));
149 assert(!Connector_Store_Idempotency_Conflict(
150 &store, "user-b", "idem-1", "digest-other"));
151 assert(Connector_Store_Audit(
152 &store, "user-a", "google:one", "gmail.send", "digest-1", 200));
153
154 Connector_Store_Close(&store);
155 }
156
157 static void test_digest(void)
158 {
159 Connector_Provider_Request request = {
160 .method = "POST",
161 .path = "/gmail/v1/users/me/messages/send",
162 .content_type = "application/json",
163 .body = "{\"raw\":\"abc\"}",
164 .body_length = 13
165 };
166 char first[65], second[65], changed[65];
167 assert(Connector_Request_Digest(
168 "user-a", "google:one", CONNECTOR_OP_GMAIL_SEND, &request, first));
169 assert(Connector_Request_Digest(
170 "user-a", "google:one", CONNECTOR_OP_GMAIL_SEND, &request, second));
171 request.body = "{\"raw\":\"xyz\"}";
172 assert(Connector_Request_Digest(
173 "user-a", "google:one", CONNECTOR_OP_GMAIL_SEND, &request, changed));
174 assert(!strcmp(first, second));
175 assert(strcmp(first, changed));
176 }
177
178 int main(void)
179 {
180 test_crypto();
181 test_store();
182 test_digest();
183 printf("core_store_test: ok\n");
184 return 0;
185 }